# +----------------------------------------------------------------------
# | ChatWork智能聊天办公系统
# +----------------------------------------------------------------------
# | 软件声明: 本系统并非自由软件,未经授权任何形式的商业使用均属非法。
# | 版权保护: 任何企业和个人不允许对程序代码以任何形式任何目的复制/分发。
# | 授权要求: 如有商业使用需求,请务必先与版权所有者取得联系并获得正式授权。
# +----------------------------------------------------------------------
# | Author: ChatWork Team <2474369941@qq.com>
# +----------------------------------------------------------------------
from pydantic import BaseModel, Field


class DrawsPromptCategoryVo(BaseModel):
    """ 绘画词库分类Vo"""
    id: int = Field(description="ID")
    name: str = Field(description="名称")

    class Config:
        json_schema_extra = {
            "example": {
                "id": 1,
                "name": "画面质感"
            }
        }


class DrawsPromptLexiconVo(BaseModel):
    """ 绘画词库词汇Vo"""
    id: int = Field(description="ID")
    prompt: str = Field(description="中文词汇")
    prompt_en: str = Field(description="英文词汇")

    class Config:
        json_schema_extra = {
            "example": {
                "id": 1,
                "prompt": "中景镜头",
                "prompt_en": "Mid shot"
            }
        }


class DrawsPromptExampleVo(BaseModel):
    """ 绘画词库示例Vo"""
    id: int = Field(description="ID")
    title: str = Field(description="标题")
    content: str = Field(description="内容")

    class Config:
        json_schema_extra = {
            "example": {
                "id": 1,
                "prompt": "月光下的沙滩",
                "prompt_en": "his photo is a beautiful beach under the moonlight..."
            }
        }
